package com.github.yangyishe.p100;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Comparator;
import java.util.List;

/**
 * 56. 合并区间
 * https://leetcode.cn/problems/merge-intervals/?envType=study-plan-v2&envId=top-interview-150
 *
 * 以数组 intervals 表示若干个区间的集合，其中单个区间为 intervals[i] = [starti, endi] 。请你合并所有重叠的区间，并返回 一个不重叠的区间数组，该数组需恰好覆盖输入中的所有区间 。
 *
 *
 *
 * 示例 1：
 *
 * 输入：intervals = [[1,3],[2,6],[8,10],[15,18]]
 * 输出：[[1,6],[8,10],[15,18]]
 * 解释：区间 [1,3] 和 [2,6] 重叠, 将它们合并为 [1,6].
 * 示例 2：
 *
 * 输入：intervals = [[1,4],[4,5]]
 * 输出：[[1,5]]
 * 解释：区间 [1,4] 和 [4,5] 可被视为重叠区间。
 *
 *
 * 提示：
 *
 * 1 <= intervals.length <= 104
 * intervals[i].length == 2
 * 0 <= starti <= endi <= 104
 */
public class Problem56 {
    public static void main(String[] args) {
        int[][] intervals=new int[][]{
                {1,4},
                {5,6}
        };

        Problem56 problem56 = new Problem56();
        int[][] merge = problem56.merge(intervals);
        for (int[] ints : merge) {
            System.out.println(Arrays.toString(ints));
        }
    }

    /**
     * 思路:
     * 1. 将参数按照起始数字进行排序
     * 2. 排序后, 进行分组遍历. 分组依据是第一个区间的开始点.
     *  如果后续区间的开始点在分组首区间的结束期限内, 则说明两个区间可以合并为一个区间, 合并规则是获取两个区间中的最大后边界, 作为合并后首区间的后边界. 删除删除(或者无视)掉当前遍历的这个区间.
     *  如果后续区间的开始点在分组首区间的结束范围后, 则说明两个元素不能合并为一个区间, 新当前遍历的区间, 作为新的分组的首区间.
     *  通过List方式记录新的区间集合, 以便能够新增区间方便
     * 3. 将获取到的List遍历为区间数组作为返回值
     *
     * 注: 如果题目已经说明区间了, 则相邻整数也不被视为一组
     * @param intervals
     * @return
     */
    public int[][] merge(int[][] intervals) {
        // 这里不知道为什么, leecode中选择使用新建Comparator排序比使用Comparator.comparing的语法要快很多
        Arrays.sort(intervals, new Comparator<int[]>() {
            @Override
            public int compare(int[] o1, int[] o2) {
                return o1[0]-o2[0];
            }
        });

        List<int[]> resultList=new ArrayList<>();
        int[] groupFirst=null;
        for (int[] interval : intervals) {
            if(groupFirst==null||groupFirst[1]<interval[0]){
                groupFirst=new int[]{interval[0],interval[1]};
                resultList.add(groupFirst);
            }else{
                groupFirst[1]=Math.max(groupFirst[1],interval[1]);
            }
        }

        return resultList.toArray(new int[resultList.size()][0]);
    }
}
